package habitant;

import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;

public class Armee {
	
	private Map<Soldat, Integer> attaque;
	private Map<Soldat, Integer> defense;
	
	public Armee(Map<Soldat, Integer> attaque, Map<Soldat, Integer> defense) {
		this.attaque = attaque;
		this.defense = defense;
	}
	
	public void setAttaque(Map<Soldat, Integer> attaque) {
		this.attaque = attaque;
	}
	public void setDefense(Map<Soldat, Integer> defense) {
		this.defense = defense;
	}

	public Map<Soldat, Integer> getAtt() {
		return attaque;
	}

	public Map<Soldat, Integer> getDef() {
		return defense;
	}
	
	public Map<Soldat, Integer> getCacAtt(){
		Map<Soldat, Integer> tmp = new HashMap<Soldat, Integer>();
		for (Entry<Soldat, Integer> entry : attaque.entrySet()){
			if (entry.getKey().get_type().equals("cac"))
				tmp.put(entry.getKey(), entry.getValue());
		}
		return tmp;
	}
	public Map<Soldat, Integer> getCacDef(){
		Map<Soldat, Integer> tmp = new HashMap<Soldat, Integer>();
		for (Entry<Soldat, Integer> entry : defense.entrySet()){
			if (entry.getKey().get_type().equals("cac"))
				tmp.put(entry.getKey(), entry.getValue());
		}
		return tmp;
		
	}
	public Map<Soldat, Integer> getDistAtt(){
		Map<Soldat, Integer> tmp = new HashMap<Soldat, Integer>();
		for (Entry<Soldat, Integer> entry : attaque.entrySet()){
			if (entry.getKey().get_type().equals("dist"))
				tmp.put(entry.getKey(), entry.getValue());
		}
		return tmp;
		
	}
	public Map<Soldat, Integer> getDistDef(){
		Map<Soldat, Integer> tmp = new HashMap<Soldat, Integer>();
		for (Entry<Soldat, Integer> entry : defense.entrySet()){
			if (entry.getKey().get_type().equals("dist"))
				tmp.put(entry.getKey(), entry.getValue());
		}
		return tmp;
		
	}
}
